Python code for length calculation